Who Should Lead Decoding Now? Tracking Reliable Trajectories for Ensembling Masked Diffusion Language Models
谁应主导解码?跟踪可靠轨迹以集成掩码扩散语言模型
🧭 掩码扩散语言模型集成与解码动态
KAIST · AITRICS
提出TIE框架,通过跟踪解码轨迹置信度动态,实现MDLM的迭代知识融合,提升推理性能
📌 工作参考价值(4 点)
- 多Agent协作推理可借鉴:跟踪各Agent中间解码状态的置信度,动态选择轨迹更可靠的Agent主导生成,提升复杂推理准确性,适用于搜索推荐中的多步推理场景。
- 生成式推荐(如Semantic ID生成)可采用类似融合策略:集成多个生成模型,在每一步去噪或解码时评估中间状态质量,由当前最优模型接管,纠正其他模型的不可靠方向。
- 搜索查询改写或广告文案生成中,可交替利用不同LLM的互补优势:当某个模型生成过程中出现低置信度区域,注入其他模型的中间状态以稳定生成轨迹。
- 该方法强调在生成过程而非结果层面融合,为电商场景中多模型协同(如多路召回融合、多生成器投票)提供新思路:关注中间表示的置信度动态而非最终输出概率。